package com.itkim.qian.utils;

import com.auth0.jwt.JWT;
import com.auth0.jwt.JWTCreator;
import com.auth0.jwt.algorithms.Algorithm;
import com.auth0.jwt.interfaces.DecodedJWT;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.util.Calendar;
import java.util.Map;

@Component
public class JWTUtil {

    private static String secret;

    @Value("${secret}")
    public void setSecret(String secret) {
        JWTUtil.secret = secret;
    }

    /**
     * 生成token
     * @param map payload数据集，含有用户所需信息
     * @return token字符串
     */
    public static String getToken(Map<String, String> map) throws Exception{
        // 设置过期时间
        Calendar instance = Calendar.getInstance();
        instance.add(Calendar.DATE,7); // 设为七天后过期

        JWTCreator.Builder builder = JWT.create();

        // 设置payload数据
        map.forEach((k,v) -> builder.withClaim(k,v));

        String token = builder.withExpiresAt(instance.getTime())
                .sign(Algorithm.HMAC256(secret));

        return token;
    }

    /**
     * 验证token的合法性
     * 由于不合法会出现异常，所以可以直接像下面的方法一样使用
     * @param token 要验证的token字符串
     * @return 返回DecodeJWT对象，该对象是已经解析出来的token的JWT对象，可以直接从里面获取出用户信息等
     */
    public static DecodedJWT verify(String token) throws Exception{
        return JWT.require(Algorithm.HMAC256(secret)).build().verify(token);
    }
}
